package com.example.chat.api;

import com.example.chat.model.*;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.beans.Transient;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@RestController
public class MessageSessionAPI {

    @Resource
    private MessageSessionMapper messageSessionMapper;
    @Resource
    private MessageMapper messageMapper;

    @GetMapping("/sessionList")
    @ResponseBody
    public Object getMessageSessionList(HttpServletRequest req) {

        List<MessageSession> messageSessionList = new ArrayList<>();

        HttpSession session = req.getSession(false);
        if (session == null) {
            System.out.println("[getMessageSessionList]: session == null");
            return messageSessionList;
        }

        User user = (User)session.getAttribute("user");
        if (user == null) {
            System.out.println("[getMessageSessionList]: user == null");
            return messageSessionList;
        }

        List<Integer> sessionIdList = messageSessionMapper.getSessionIdsByUserId(user.getUserId());
        for (int sessionId : sessionIdList) {
            List<Friend> friends = messageSessionMapper.getFriendsBySessionId(sessionId, user.getUserId());
            String lastMessage = messageMapper.getLastMessageBySessionId(sessionId);

            MessageSession messageSession = new MessageSession();
            messageSession.setSessionId(sessionId);
            messageSession.setFriends(friends);
            if (lastMessage == null) {
                messageSession.setLastMessage("");
            } else {
                messageSession.setLastMessage(lastMessage);
            }

            messageSessionList.add(messageSession);
        }

        return messageSessionList;
    }

    @PostMapping("/session")
    @ResponseBody
    @Transactional // 事务，使操作原子化，一体化
    public Object addMessageSession(int toUserId, HttpServletRequest req) {

        HashMap<String, Integer> resp = new HashMap<>();

        HttpSession session = req.getSession(false);
        if (session == null) {
            System.out.println("[addMessageSession]: session == null");
            return null;
        }

        User user = (User)session.getAttribute("user");
        if (user == null) {
            System.out.println("[addMessageSession]: user == null");
            return null;
        }

        MessageSession messageSession = new MessageSession();
        messageSessionMapper.addMessageSession(messageSession);

        MessageSessionUserItem messageSessionUserItem1 = new MessageSessionUserItem();
        messageSessionUserItem1.setSessionId(messageSession.getSessionId());
        messageSessionUserItem1.setUserId(user.getUserId());

        messageSessionMapper.addMessageSessionUser(messageSessionUserItem1);


        MessageSessionUserItem messageSessionUserItem2 = new MessageSessionUserItem();
        messageSessionUserItem2.setSessionId(messageSession.getSessionId());
        messageSessionUserItem2.setUserId(toUserId);

        messageSessionMapper.addMessageSessionUser(messageSessionUserItem2);

        System.out.println("[addMessageSession] 新增会话成功！ sessionId = " + messageSession.getSessionId()
                + " userId1 = " + user.getUserId() + " userId2 = " + toUserId);

        resp.put("sessionId", messageSession.getSessionId());

        return resp;
    }
}
